from django.shortcuts import render
from rest_framework import viewsets
from .models import School, College, Team, ResearchGroup
from .serializers import SchoolSerializer, CollegeSerializer, TeamSerializer, ResearchGroupSerializer


class SchoolViewSet(viewsets.ModelViewSet):
    """
    SchoolInfo
    """
    queryset = School.objects.all()
    serializer_class = SchoolSerializer


class CollegeViewSet(viewsets.ModelViewSet):
    """
    InstituteInfo
    """
    queryset = College.objects.all()
    serializer_class = CollegeSerializer


class TeamViewSet(viewsets.ModelViewSet):
    """
    TeamInfo
    """
    queryset = Team.objects.all()
    serializer_class = TeamSerializer


class ResearchGroupViewSet(viewsets.ModelViewSet):
    """
    ResearchGroupInfo
    """
    queryset = ResearchGroup.objects.all()
    serializer_class = ResearchGroupSerializer
